Skip to content

Conversation

@boltomli
Copy link

@boltomli boltomli commented Nov 21, 2025

Resolve issue #351 #357 and alike.

@facundoq
Copy link

facundoq commented Dec 3, 2025

This fixes the issue for CPUs but a better way to do this is to actually use the device parameter for the map_location argument to torch.load().

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants